Добавляет форму поиска в главное (или любое другое) меню. Какое именно - определяется menu_id == 'nav'

add_filter('wp_nav_menu_items', 'search_box_function', 10, 2);
function search_box_function( $nav, $args ) {
    if( $args->menu_id == 'nav' ) {
        $nav .= "